from transformers import AutoModelForCausalLM, AutoTokenizer
model_path="deepseek-ai/DeepSeek-R1"
tokenizer =AutoTokenizer.from_pretrained(model_path)
model = AutoModelForCausalLM.from_pretrained(model_path,device_map="auto")